package com.huanzhidadi.single.proxy;

import org.springframework.cglib.proxy.*;

import java.lang.reflect.Method;
import java.util.Date;

/**
 * Cglib动态代理，基于ASM机制实现
 */
public class CglibProxyTest {
    public static void main(String[] args) {
        ProxyInterceptor proxyInterceptor = new ProxyInterceptor();
        LogInterceptor logInterceptor = new LogInterceptor();

        // 字节码增强器,它可以方便的对你想要处理的类进行扩展
        // 动态的创建给定类的子类并且拦截代理类的所有的方法
        Enhancer enhancer = new Enhancer();
        // 设置超类，cglib是通过继承来实现的
        enhancer.setSuperclass(TargetObject.class);
        // 设置多个拦截器，NoOp.INSTANCE是一个空拦截器，不做任何处理
        enhancer.setCallbacks(new Callback[]{proxyInterceptor, logInterceptor, NoOp.INSTANCE});
        // 设置回调过滤器
        enhancer.setCallbackFilter(new ProxyFilter());
        // 创建代理类实例
        TargetObject targetObject = (TargetObject) enhancer.create();
        targetObject.say();
        targetObject.work();
    }
}

// 被代理类
class TargetObject {
    public void say() {
        System.out.println("被代理类 say");
    }

    public void work() {
        System.out.println("被代理类 work");
    }
}

/**
 * 继承 MethodInterceptor，用于方法的拦截回调
 */
class ProxyInterceptor implements MethodInterceptor {
    /**
     * @param object      表示要进行增强的对象
     * @param method      表示拦截的方法
     * @param args     数组表示参数列表，基本数据类型需要传入其包装类型，如int-->Integer、long-Long、double-->Double
     * @param methodProxy 表示对方法的代理，invokeSuper方法表示对被代理对象方法的调用
     * @return 执行结果
     * @throws Throwable
     */
    @Override
    public Object intercept(Object object, Method method, Object[] args, MethodProxy methodProxy) throws Throwable {
        System.out.println("代理类 say hello");
        // 注意这里是调用 invokeSuper 而不是 invoke，否则死循环，methodProxy.invokesuper执行的是原始类的方法，method.invoke执行的是子类的方法
        Object result = methodProxy.invokeSuper(object, args);
        return result;
    }
}

/**
 * 可以指定多个 MethodInterceptor 进行过滤筛选
 */
class LogInterceptor implements MethodInterceptor {
    /**
     * @param object      表示要进行增强的对象
     * @param method      表示拦截的方法
     * @param args     数组表示参数列表，基本数据类型需要传入其包装类型，如int-->Integer、long-Long、double-->Double
     * @param methodProxy 表示对方法的代理，invokeSuper方法表示对被代理对象方法的调用
     * @return 执行结果
     * @throws Throwable
     */
    @Override
    public Object intercept(Object object, Method method, Object[] args, MethodProxy methodProxy) throws Throwable {
        before();
        Object result = methodProxy.invokeSuper(object, args);
        after();
        return result;
    }
    private void before() {
        System.out.println(String.format("log start time [%s] ", new Date()));
    }
    private void after() {
        System.out.println(String.format("log end time [%s] ", new Date()));
    }
}

/**
 * 回调过滤器: 在CGLib回调时可以设置对不同方法执行不同的回调逻辑，或者根本不执行回调。
 */
class ProxyFilter implements CallbackFilter {
    @Override
    public int accept(Method method) {
        if ("work".equals(method.getName())) {
            return 1;   // Callback 列表第2个拦截器
        }
        return 0;   // Callbacks 列表第1个拦截器，return 2 则为第3个，以此类推
    }
}